#-*- coding: utf-8 -*-
from tutor.scripts import *
from tutor.plugins.sympy import *

while True:
    # Entradas
    with display_block('Entradas'):
        R = together(sqrt(oneof(1, 2, 3, 4, 6, 8) / Two))
        a = together(oneof(1, 2, 3, 4, 6) / Two)
        rho = 1 + r**a
    
    # Respostas
    with display_block('Respostas (questão 1)'):
        res = 2*pi*(R**2/2 + R**(a+2)/(a+2))
        correta = together(res.expand())
        
        # Usar apenas expressões mais simples
        if type(correta) is Add:
            continue
    
        res = 2*pi*(R + R**(a+1)/(a+1))
        jacobiano = together(res.expand())
    
        res = 2*pi*(R**2/2 + R**(a+1)/(a+1))
        dis_1 = together(res.expand())
    
        res = 2*pi*(R**3/3 + R**(a+3)/(a+3))
        dis_2 = together(res.expand())
        dis_3 = jacobiano * 2
        
        # Inercia
        res = 2*pi*(R**4/4 + R**(a+4)/(a+4))
        inercia = dis_4 = together(res.expand())
        del res
        
    with display_block('Respostas (questão 2)'):
        idis_1 = dis_1
        idis_2 = dis_2
        idis_3 = dis_3
        idis_4 = correta
        idis_5 = jacobiano 
    break